import torch

A = torch.arange(16)
print(A)
B = A.view(4, 4)
print(B)
C = A.reshape(2, 8)
print(C)
print("-"*20)
B = A.view(1, 2, 1, 8)
print(B)
C = B.squeeze()
D = C.unsqueeze(0)
E = C.unsqueeze(1)
print(C)
print(D)
print(E)
F =  C.unsqueeze(2)
print(F)
